Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Weights and Biases Callback for Elegy #220

Merged
merged 39 commits into from
Mar 23, 2022
Merged

Weights and Biases Callback for Elegy #220

merged 39 commits into from
Mar 23, 2022

Conversation

soumik12345
Copy link
Contributor

Callback that streams epoch results to a Weights & Biases run.

Sample Usage:

wandb.login()
wandb_logger = WandbCallback(project="sample-wandb-project", job_type="train")
model.fit(X_train, Y_train, callbacks=[wandb_logger])

Key Features:

  • Automatically log your experiment results using elegy.callbacks.WandbCallback
  • Save and version model checkpoints as Weights & Biases Artifacts

Colab Notebook for Demonstration: https://colab.research.google.com/drive/1M-KJcxFx5LXVqtt7Yztbf8uyX0Cd7GEw?usp=sharing
Sample Dashboard: https://wandb.ai/geekyrakshit/elegy-test

@soumik12345
feature: added wandb callback
6f2be70
@soumik12345
chore: added WandbCallback import inside __init__.py
3ba4ce3
@soumik12345
chore: applied black
08808ea
@soumik12345
chore: added documentation for WandbCallback
90c6b21
@soumik12345
chore: updated wandb run initialization + added run finish on ending …
1d7cedf 
…training
@soumik12345
updated poetry dependencies
cb7fcf1
@soumik12345
updated run initialization in wandb callback
b7ff92c
@soumik12345
fixec wandb import
ebb945e
@soumik12345
fixed on_train_end in wandb callback
2d1210a
@soumik12345
added 'size' to ignore fields + added 'train' alias to train metrics
aac79bd
@soumik12345
fix: wandb run keys
f423035
@soumik12345
fix: updated run key names
dfc2a3a
@soumik12345
minor bug fix
cad4ae5
@soumik12345
feature: instead of logging constant metrics step-wise, logs them to …
90bacff 
…the run summary
@soumik12345
updated wandb sun summary
db07abe
@soumik12345
updated configs
40609df
@soumik12345
updated callback
5e08aaf
@soumik12345
updated callback
03bb72e
@soumik12345
added method to gather module attributes
d1abc88
@soumik12345
fix: fixed minor bug in gathering model attributes
0cf6479
@soumik12345
updated callback example
78d35e1
@soumik12345
added train alias
b21f38b
@soumik12345
Minor Bug Fix
3c8babc
@soumik12345
updated wandb callback signature
10e177a
@soumik12345
fixed typo
3f112b3
@soumik12345
Added model checkpointing
4d08a48
@soumik12345
updated checkpoint system
8929139
@soumik12345
updated checkpoint logic
494942e
@soumik12345
ipdated checkpoint system
9522109
@soumik12345
Merge pull request #1 from soumik12345/checkpoints
f7dca2f 
Artifact-based Checkpoint Storage
@cgarciae cgarciae merged commit c6ad092 into poets-ai:master Mar 23, 2022
@cgarciae
Copy link
Collaborator

Hey @soumik12345! Sorry it took so long to review.
I updated the dependencies and solved merge issue.

Thanks a lot for doing this ❤️

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
feature
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@soumik12345 @cgarciae